package com.bang.study.day04;

/**
 * @Auther: Bang
 * @Date: 2019/9/16 0016 17:22
 * @Description: 307. 区域和检索 - 数组可修改
 * 给定一个整数数组  nums，求出数组从索引 i 到 j  (i ≤ j) 范围内元素的总和，包含 i,  j 两点。
 *
 * update(i, val) 函数可以通过将下标为 i 的数值更新为 val，从而对数列进行修改。
 *
 * 示例:
 *
 * Given nums = [1, 3, 5]
 *
 * sumRange(0, 2) -> 9
 * update(1, 2)
 * sumRange(0, 2) -> 8
 * 说明:
 *
 * 数组仅可以在 update 函数下进行修改。
 * 你可以假设 update 函数与 sumRange 函数的调用次数是均匀分布的。
 */
public class Solution_307_seg {
    /**
     * 基于线段树
     */

    class NumArray {

        private SegmentTree<Integer> segmentTree;


        public NumArray(int[] nums) {
            if (nums.length > 0) {
                Integer[] arrs = new Integer[nums.length];
                for (int i = 0; i < nums.length; i++) {
                    arrs[i] = nums[i];
                }
                segmentTree = new SegmentTree<Integer>(arrs, new Merger<Integer>() {
                    @Override
                    public Integer merger(Integer e1, Integer e2) {
                        return e1 + e2;
                    }
                });
            }
        }

        public void update(int index, int val) {
            if (null==segmentTree){
                throw new RuntimeException("segmentTree is null");
            }
            segmentTree.set(index, val);

        }

        public int sumRange(int i, int j) {
            return segmentTree.query(i, j);
        }
    }

    public static void main(String[] args) {
        int[] arr = {1,2,3};
        NumArray numArray = new Solution_307_seg().new NumArray(arr);
        int i = numArray.sumRange(0, 1);
        System.out.println(i);
    }
}
